from selenium.webdriver.support.wait import WebDriverWait
from appium.webdriver.common.appiumby import By

class CartSettlement:
    def cart_settlement(self, driver):
        WebDriverWait(driver, 10).until(lambda x: x.find_element(by=By.ID, value="com.tpshop.malls:id/bottom_cart_img")).click()
        WebDriverWait(driver, 10).until(lambda x: x.find_element(by=By.ID, value="com.tpshop.malls:id/buy_tv")).click()